// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef NET_SOCKET_WEBSOCKET_TRANSPORT_CONNECT_SUB_JOB_H_
#define NET_SOCKET_WEBSOCKET_TRANSPORT_CONNECT_SUB_JOB_H_

#include <stddef.h>

#include <memory>
#include <utility>

#include "base/compiler_specific.h"
#include "base/macros.h"
#include "net/base/address_list.h"
#include "net/base/load_states.h"
#include "net/socket/websocket_endpoint_lock_manager.h"
#include "net/socket/websocket_transport_client_socket_pool.h"

namespace net {

class BoundNetLog;
class ClientSocketFactory;
class IPEndPoint;
class StreamSocket;

// Attempts to connect to a subset of the addresses required by a
// WebSocketTransportConnectJob, specifically either the IPv4 or IPv6
// addresses. Each address is tried in turn, and parent_job->OnSubJobComplete()
// is called when the first address succeeds or the last address fails.
class WebSocketTransportConnectSubJob
    : public WebSocketEndpointLockManager::Waiter {
public:
    typedef WebSocketTransportConnectJob::SubJobType SubJobType;

    WebSocketTransportConnectSubJob(const AddressList& addresses,
        WebSocketTransportConnectJob* parent_job,
        SubJobType type);

    ~WebSocketTransportConnectSubJob() override;

    // Start connecting.
    int Start();

    bool started() { return next_state_ != STATE_NONE; }

    LoadState GetLoadState() const;

    SubJobType type() const { return type_; }

    std::unique_ptr<StreamSocket> PassSocket()
    {
        return std::move(transport_socket_);
    }

    // Implementation of WebSocketEndpointLockManager::EndpointWaiter.
    void GotEndpointLock() override;

private:
    enum State {
        STATE_NONE,
        STATE_OBTAIN_LOCK,
        STATE_OBTAIN_LOCK_COMPLETE,
        STATE_TRANSPORT_CONNECT,
        STATE_TRANSPORT_CONNECT_COMPLETE,
        STATE_DONE,
    };

    ClientSocketFactory* client_socket_factory() const;

    const BoundNetLog& net_log() const;

    const IPEndPoint& CurrentAddress() const;

    void OnIOComplete(int result);
    int DoLoop(int result);
    int DoEndpointLock();
    int DoEndpointLockComplete();
    int DoTransportConnect();
    int DoTransportConnectComplete(int result);

    WebSocketTransportConnectJob* const parent_job_;

    const AddressList addresses_;
    size_t current_address_index_;

    State next_state_;
    const SubJobType type_;

    std::unique_ptr<StreamSocket> transport_socket_;

    DISALLOW_COPY_AND_ASSIGN(WebSocketTransportConnectSubJob);
};

} // namespace net

#endif // NET_SOCKET_WEBSOCKET_TRANSPORT_CONNECT_SUB_JOB_H_
